#!/usr/bin/python3
# ******************************************************************************
# Copyright (c) Huawei Technologies Co., Ltd. 2021-2021. All rights reserved.
# licensed under the Mulan PSL v2.
# You can use this software according to the terms and conditions of the Mulan PSL v2.
# You may obtain a copy of Mulan PSL v2 at:
#     http://license.coscl.org.cn/MulanPSL2
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR
# PURPOSE.
# See the Mulan PSL v2 for more details.
# ******************************************************************************/
import kafka
from kafka.errors import KafkaError

from python_learn.log.logger import log


class Producer:
    def __init__(self):
        self.producer = kafka.KafkaProducer(bootstrap_servers="localhost:9092")

    def send_msg(self, body=None):
        future = self.producer.send(topic='test', key=b'1', value=b'test_msg')
        try:
            response = future.get(timeout=10)
            print(response.topic)
            print(response.partition)
            print(response.offset)
        except KafkaError as e:
            log.error("Send error msg is %s" % e)


if __name__ == '__main__':
    producer = Producer()
    producer.send_msg()
